package com.pj.project.mqtt.myserver;

import com.pj.project.update_package.UpdatePackageService;
import com.pj.utils.sg.NbUtil;
import lombok.extern.slf4j.Slf4j;
import org.eclipse.paho.client.mqttv3.*;
import org.eclipse.paho.client.mqttv3.persist.MemoryPersistence;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.nio.charset.StandardCharsets;

import static com.pj.project.mqtt.util.Tools.bytesToHex;

/**
 * 消息订阅
 * @param
 * @throws MqttException
 */
@Slf4j
@Component
public class SubscribeSample {
    @Autowired
    UpdatePackageService updatePackageService;

    public static MqttClient client;




//    @PostConstruct
    public void samp() throws MqttException {
//        String HOST = "tcp://39.99.255.56:1883";
        String HOST = "tcp://mqtt.zyjzjs.cn:1883";
//        String TOPIC = "$dp";
        String TOPIC = "device/update/rx/+";
        int qos = 0;
        String clientid = "mqtt-server--";
        String userName = "admin";
        String passWord = "admin";
        try {
            String randomString = NbUtil.getRandomString(5);
            // host为主机名，test为clientid即连接MQTT的客户端ID，一般以客户端唯一标识符表示，MemoryPersistence设置clientid的保存形式，默认为以内存保存
            client = new MqttClient(HOST, clientid + randomString, new MemoryPersistence());
            // MQTT的连接设置
            MqttConnectOptions options = new MqttConnectOptions();
            // 设置是否清空session,这里如果设置为false表示服务器会保留客户端的连接记录，这里设置为true表示每次连接到服务器都以新的身份连接
            options.setCleanSession(true);
            // 设置连接的用户名
            options.setUserName(userName);
            // 设置连接的密码
            options.setPassword(passWord.toCharArray());
            // 设置超时时间 单位为秒
            options.setConnectionTimeout(10);
            // 设置会话心跳时间 单位为秒 服务器会每隔1.5*20秒的时间向客户端发送个消息判断客户端是否在线，但这个方法并没有重连的机制
            options.setKeepAliveInterval(20);
            // 设置回调函数
            client.setCallback(new MqttCallback() {

                public void connectionLost(Throwable throwable) {
                    log.error("断开连接！");
                    try {
                        client.connect(options);
                    } catch (MqttException e) {
                        log.error("重连失败！");
                        try {
                            client.connect(options);
                        } catch (MqttException ex) {
                            log.error("2次重连失败！");

                        }
                    }
                }

                public void messageArrived(String topic, MqttMessage message) throws Exception {
                    String replace = topic.replace("/rx/", "/tx/");
                    updatePackageService.updatePackageCtl(replace, message.getPayload());
                }

                public void deliveryComplete(IMqttDeliveryToken token) {
//                    System.out.println("deliveryComplete---------" + token.isComplete());
                }
            });
            client.connect(options);
            // 订阅消息
            client.subscribe(TOPIC, qos);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void sendDataToMqtt(String topic,int data){
        MqttMessage message = new MqttMessage(intToByteArray(data));
        message.setQos(2);
        try {
            log.info("发送给:"+ topic +"消息："+ bytesToHex(message.getPayload()));
            client.publish(topic, message);
        } catch (MqttException e) {
        }
    }

    public void sendDataToMqtt(String topic,byte[] data){
        MqttMessage message = new MqttMessage(data);
//		message.setQos(0);
        try {
//			log.info("发送给:"+ topic +"消息："+ bytesToHex(message.getPayload()));
            client.publish(topic, message);
        } catch (MqttException e) {
            log.info(e.getMessage());
        }
    }

    public void sendDataToMqtt(String topic,String data){
        MqttMessage message = new MqttMessage(data.getBytes(StandardCharsets.UTF_8));
        message.setQos(2);
        try {
            log.info("发送给:"+ topic +"消息："+ bytesToHex(message.getPayload()));
            client.publish(topic, message);
        } catch (MqttException e) {
        }
        System.out.println(bytesToHex(message.getPayload()));
    }

    public static byte[] intToByteArray(int i) {
        byte[] result = new byte[4];
        result[0] = (byte)((i >> 24) & 0xFF);
        result[1] = (byte)((i >> 16) & 0xFF);
        result[2] = (byte)((i >> 8) & 0xFF);
        result[3] = (byte)(i & 0xFF);
        return result;
    }

}
